from langchain_openai import ChatOpenAI


def get_wantong_gpt4(model_name="gpt-4-turbo-2024-04-09"):
    """
    openai
    :param model_name:
    :return:
    """
    # return azure_chat_openai("openai-gpt4")
    llm = ChatOpenAI(
        openai_api_base="https://api.ai-gaochao.cn/v1",
        openai_api_key="sk-XYg12wuT0f5EVhsF7780B28d06Fb4053A209970cDc146598",
        model_name=model_name
    )
    return llm


def get_deepseek(model_name="deepseek-chat"):
    """
    openai
    :param model_name:
    :return:
    """
    # return azure_chat_openai("openai-gpt4")
    llm = ChatOpenAI(
        openai_api_base="https://api.deepseek.com/v1",
        openai_api_key="sk-0f22ac7eb510460eae4185a4e854e78c",
        model_name=model_name
    )
    return llm


if __name__ == '__main__':
    llm = get_deepseek()
    res = llm.invoke("您好")
    print(res)
